import numpy as np

import shared_dataset as sh


def create_data():
    brs = np.array([[3, 1, 2, 4], [3, 2, 1, 6], [6, 7, 1, 5]])

    ms = np.array([[0, 1, 0, 1], [0, 1, 1, 1], [1, 1, 0, 0]])

    sfs = np.array([[1, 2], [1, 2], [1]],dtype=object)

    np.savez('test.npz', brs=brs, ms=ms, sfs=sfs)


def main():
    (en_train, ms_train, de_train, y_train), (en_test, ms_test, de_test, y_test) = sh.load_data(path='test.npz',
                                                                                                num_words=10000,
                                                                                                num_sfs=1003)

    print(en_train)

    print(ms_train)

    print(de_train)

    print(y_train)

    print('--------------------------------')

    print(en_test)
    print('--------------------------------')
    print(ms_test)
    print('--------------------------------')
    print(de_test)
    print('--------------------------------')
    print(y_test)


if __name__ == '__main__':
    create_data()

    main()